package date_202404

//42. 接雨水
//已解答
//困难
//相关标签
//相关企业
//给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。
//
//
//
//示例 1：
//
//
//
//输入：height = [0,1,0,2,1,0,1,3,2,1,2,1]
//输出：6
//解释：上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。
//示例 2：
//
//输入：height = [4,2,0,3,2,5]
//输出：9
//
//
//提示：
//
//n == height.length
//1 <= n <= 2 * 104
//0 <= height[i] <= 105

func trap(height []int) int {
	if len(height) <= 2 {
		return 0
	}
	leftMax := make([]int, len(height))
	rightMax := make([]int, len(height))
	leftMax[0] = height[0]
	rightMax[len(height)-1] = height[len(height)-1]
	for i := 1; i < len(height); i++ {
		leftMax[i] = IntMax1(leftMax[i-1], height[i])
		rightMax[len(height)-i-1] = IntMax1(height[len(height)-i-1], rightMax[len(height)-i])
	}

	trapVal := 0
	for i := 1; i < len(height)-1; i++ {
		leftMaxVal := leftMax[i-1]
		rightMaxVal := rightMax[i+1]
		minVal := IntMin1(leftMaxVal, rightMaxVal)
		if height[i] < minVal {
			trapVal = trapVal + minVal - height[i]
		}
	}
	return trapVal
}

func IntMax1(a, b int) int {
	if a > b {
		return a
	}
	return b
}

func IntMin1(a, b int) int {
	if a < b {
		return a
	}
	return b
}
